# server.py
from mcp.server.fastmcp import FastMCP
import sys
import logging
import requests
import json
print('start0')
introduction='''
星辰科技集团（StellarTech Group）公司简介

公司概况
星辰科技集团（StellarTech Group）成立于2010年，是一家以科技创新为核心驱动力的全球化高科技企业，总部位于中国深圳，并在美国硅谷、德国慕尼黑、日本东京及新加坡设有研发中心与分支机构。公司专注于人工智能、云计算、物联网（IoT）、大数据分析及智能制造等前沿技术领域，致力于为全球客户提供领先的数字化解决方案。

自创立以来，星辰科技集团秉持“创新引领未来，科技改变世界”的企业使命，以卓越的技术实力和敏锐的市场洞察力，迅速成长为行业内的标杆企业。目前，公司拥有员工超过5000人，其中研发人员占比超过60%，已获得国内外专利技术800余项，业务覆盖全球50多个国家和地区，服务客户包括世界500强企业、政府机构及新兴科技公司。

核心业务与产品
1. 人工智能与大数据
星辰科技集团在人工智能领域深耕多年，自主研发的“星智（StellarMind）”AI平台集成了机器学习、自然语言处理（NLP）、计算机视觉等先进技术，广泛应用于金融、医疗、零售及智慧城市等行业。例如，在医疗领域，公司的AI辅助诊断系统已帮助数百家医院提升诊疗效率；在金融行业，智能风控系统为银行和保险机构提供精准的数据分析与决策支持。

2. 云计算与企业服务
依托强大的云计算基础设施，星辰科技推出了“星云（StellarCloud）”企业级云服务平台，为客户提供弹性计算、数据存储、安全防护等一站式云解决方案。无论是中小型企业还是大型集团，均可通过星云平台实现业务的高效上云与智能化管理。

3. 物联网与智能制造
在工业4.0时代，星辰科技通过物联网技术赋能传统制造业，打造了“星联（StellarLink）”工业互联网平台，实现设备互联、远程监控及智能运维。公司已为汽车制造、电子设备、能源等行业提供定制化智能制造方案，助力企业降本增效。

4. 智慧城市与可持续发展
作为智慧城市建设的积极参与者，星辰科技利用大数据和AI技术优化城市管理，开发了智慧交通、环境监测、公共安全等解决方案。同时，公司积极响应全球碳中和目标，推动绿色科技研发，如智能能源管理系统和可再生能源优化技术。

企业文化与价值观
星辰科技集团的企业文化以“创新、协作、责任、卓越”为核心价值观：

创新（Innovation）：鼓励员工突破思维边界，持续探索技术前沿。

协作（Collaboration）：倡导跨部门、跨地区的团队合作，共享智慧。

责任（Responsibility）：坚持科技向善，注重数据安全与社会责任。

卓越（Excellence）：追求极致的产品质量与客户体验。

公司高度重视人才培养，设立了“星辰学院”内部培训体系，并与全球顶尖高校合作开展技术研究。此外，星辰科技积极履行社会责任，通过公益基金会支持教育、环保和科技普惠项目。

全球布局与未来愿景
星辰科技集团始终坚持全球化战略，通过本地化运营与国际化团队，深度融入全球市场。未来五年，公司计划进一步扩大在东南亚、欧洲及北美市场的投入，并加强在量子计算、边缘计算等新兴领域的研发。

愿景：成为全球最具影响力的科技企业，用创新技术推动人类社会进步。
使命：以科技之力，连接智慧未来。

星辰科技集团，与您携手，共创数字新时代！
'''

def ai(question,background):
    url = "https://api.siliconflow.cn/v1/chat/completions"

    payload = {
        "model": "Qwen/Qwen2.5-7B-Instruct",
        "messages": [
                    {
                "role": "assistant",
                "content": f"背景资料："+background
            },
                    {
                "role": "system",
                "content": f"根据背景资料回答用户的问题，回答要简短。"
            },
            {
                "role": "user",
                "content": question
            }
        ]
    }
    headers = {
        "Authorization": "Bearer sk-zkqxvkdkhwkahwrydvjigsqkkykhaxedpmorwucaemypswid",
        "Content-Type": "application/json"
    }
    print('start1')
    response = requests.request("POST", url, json=payload, headers=headers)
    answer=response.json()["choices"][0]['message']['content']
    return answer

logger = logging.getLogger('log')

# Fix UTF-8 encoding for Windows console
if sys.platform == 'win32':
    sys.stderr.reconfigure(encoding='utf-8')
    sys.stdout.reconfigure(encoding='utf-8')


import requests



# Create an MCP server·12
mcp = FastMCP("mcps")

# Add an addition tool
# 修改原有的ZhipuAI初始化
@mcp.tool()
def 请问(query_text: str) -> list:


    content_list=ai(query_text,introduction)
    
    logger.info(f"联网查询完成，查询内容: {query_text} and {content_list}")
    return {"success": True, "result": content_list}





# Start the server
if __name__ == "__main__":
    mcp.run(transport="stdio")


def load_config():
    with open('config.json', 'r', encoding='utf-8') as f:  # Add encoding
        return json.load(f)

def save_log(message):
    with open('query.log', 'a', encoding='utf-8') as f:  # Add encoding
        f.write(f"{datetime.now()} - {message}\n")
